package com.lzp.leetcode

/**
 * https://leetcode.cn/problems/two-sum/description/
 * @describe: 两数之和
 * @Author: lixiaopeng
 * @Date: 2023/1/30
 *
给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
 */

fun main() {
    val intArrayOf = intArrayOf(3, 2, 4)
    println(intArrayOf.contentToString())
    val twoSum = twoSum1(intArrayOf, 6)
    println(twoSum.contentToString())
}

/**
 * 优化版
 */
fun twoSum2(nums: IntArray, target: Int): IntArray{
    val map = hashMapOf<Int, Int>()
    val res = ArrayList<Int>()
    for (i in nums.indices) {
        if (map.containsKey(target - nums[i])) {
            res.add(i)
            res.add(map[target - nums[i]]!!)
        } else {
            map[nums[i]] = i
        }
    }
    return res.toIntArray()
}

fun twoSum1(nums: IntArray, target: Int): IntArray {
    if (nums.size < 2) {
        return intArrayOf(0, 0)
    }

    for (index in nums.indices) {
        for (j in index + 1 until nums.size) {
            if (nums[index] + nums[j] == target) {
                return intArrayOf(index, j)
            }
        }
    }
    return intArrayOf(0, 0)
}
